package bfs;

import java.util.Arrays;
import java.util.LinkedList;
import java.util.Queue;

//正难则反：
//   从正面考虑，将所有的1入队列，但问题是bfs只有相邻层的信息，如果一个1到0的最小距离>2,那么
//   起点就会丢失，导致无法进行；
//   从反面考虑，将所有的0入队列，0不需要考虑起始坐标，只需要一直往外扩就行，遇到1距离直接就为bfs的
//   扩散层数或者上一层的距离+1,因此可以完成任务；
public class zo矩阵 {
	public static int[] dx={1,-1,0,0};
    public static int[] dy={0,0,1,-1};
    public static boolean[][] visit;
    public int[][] updateMatrix(int[][] mat) {
        int m=mat.length;
        int n=mat[0].length;
        visit=new boolean[m][n];
        //记录最小距离的矩阵
        int[][] dist=new int[m][n];
        //一开始全部初始化为-1，表示当前还未填入最小距离
        for(int i=0;i<m;i++) Arrays.fill(dist[i],-1);
        Queue<int[]> q=new LinkedList<>();
        //将所有的0入队列并更新距离
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                if(mat[i][j]==0){
                    q.offer(new int[]{i,j});
                    visit[i][j]=true;
                    dist[i][j]=0;
                }
            }
        }

        //bfs
        while(!q.isEmpty()){
            int sz=q.size();
            for(int i=0;i<sz;i++){
                int[] tmp=q.poll();
                int nx=tmp[0],ny=tmp[1];
                //向外层扩散
                for(int k=0;k<4;k++){
                    int x=nx+dx[k],y=ny+dy[k];
                    //如果一个位置未填入最小值，则需要扩散，否则不扩散
                    if(x>=0&&x<m&&y>=0&&y<n&&visit[x][y]==false&&dist[x][y]==-1){
                    	//由于0早已被填入，因此到这mat[x][y]一定=1
                    	//(nx,ny) -> (x,y)，因此直接在(nx,ny)的距离处+1
                        dist[x][y]=dist[nx][ny]+1;
                        visit[x][y]=true;
                        q.offer(new int[]{x,y});
                    }
                }
            }
        }
        return dist;
    }
}
